#ifndef CUDA_POISSON_H
#define CUDA_POISSON_H

#include "CUDA_vec.h"

class CUDA_Poisson {
public:
	CUDA_Poisson(bool use_sh){ use_shmem = use_sh;}

	void operator()(const CUDA_vec<float> & a, CUDA_vec<float> & b) const;

	void getDiagonal(CUDA_vec<float> & a) const;
private:
	bool use_shmem;
};

#endif
